package 高效制胜._02;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * @author yzh
 * @version 1.0
 * @date 2021/9/8 18:31
 * 四数之和
 * 算法：排序  双指针
 * 这里的 target 没有指定 0
 * 所以 nums[i] > target 不能跳出循环
 */
public class _18 {
    public static void main(String[] args) {
        int[] nums = {1, -2, -5, -4, -3, 3, 3, 5};
        // -5 -4 -3 -2 1 3 3 5
        System.out.println(new _18().fourSum(nums, -11));
    }

    public List<List<Integer>> fourSum(int[] nums, int target) {
        List<List<Integer>> ans = new ArrayList<>();
        Arrays.sort(nums);
        int n = nums.length;
        for (int i = 0; i < n - 3; i++) {
            if (i != 0 && nums[i] == nums[i - 1]) continue;
            for (int j = i + 1; j < n - 2; j++) {
                if (j != i + 1 && nums[j] == nums[j - 1]) continue;
                int l = j + 1, r = n - 1;
                while (l < r) {
                    if (l != j + 1 && nums[l] == nums[l - 1]) {
                        ++l;
                        continue;
                    }
                    if (r != n - 1 && nums[r] == nums[r + 1]) {
                        --r;
                        continue;
                    }
                    int sum = nums[i] + nums[j] + nums[l] + nums[r];
                    if (sum == target) {
                        ans.add(Arrays.asList(nums[i], nums[j], nums[l], nums[r]));
                        ++l;
                        --r;
                    } else if (sum > target) --r;
                    else ++l;
                }
            }
        }
        return ans;
    }

}
